from api.ihrmEmpManageApi import EmpManageApi
from common.assertTools import common_assert

from common.db_tools import DBTools
from common.getHeader import get_header
from config import EMP_ID


class TestQueryEmp(object):
    def setup_class(self):
        # 创建实例
        self.emp = EmpManageApi()
        # 准备请求头
        self.req_header = get_header()
    def setup(self):

        # 添加一个员工id 这个员工id会在后续的查询员工接口中使用
        DBTools.db_iud(F"insert into bs_user(id,mobile,username) values('{EMP_ID}','13200340089','张大');")

    def teardown(self):
        # 测试结束后删除上面插入的那条数据
        DBTools.db_iud(F"delete from bs_user where id='{EMP_ID}';")
    # 查询成功
    def test1_query_success(self):

        resp = self.emp.query_emp(self.req_header, EMP_ID)
        print("查询员工成功:", resp.json())
        # 断言
        common_assert(resp, 200, True, 10000, "操作成功")
    # id不存在
    def test2_id_no_exists(self):

        resp = self.emp.query_emp(self.req_header, "1234567")
        print("id不存在:", resp.json())
        # 断言
        common_assert(resp, 200, False, 20005, "查询员工失败")
